#include <iostream>
#include "BaseObject.h"
#include "calvin.h"
#include "Wallace-Gromit.h"


BaseObject * menu();


int main (int argc, char * const argv[]) {
    // insert code here...
    BaseObject * myobj= menu();
    
	myobj->Draw();
    return 0;
}

BaseObject * menu(){
	BaseObject * result = NULL;
	int input;
	std::cout << "Choose 1 or 2\n";
	std::cin >> input;
	switch (input){
		case 1:
			result = new Wallace_GromitObject();
			break;
		case 2:
			result = new calvinObject();
			break;
		default:
			result = new Wallace_GromitObject;
		}
	return result;
}